from turtle import *
from random import randrange
import time

def draw_square(x, y, size, color_name):
    '''
    画正方形
    :param x 表示在画布的x位置
    :param y 表示画布的y位置
    :param size 表示正方形的长度
    :param color_name 表示正方形的颜色
    :return
    '''
    ht()    # 是海龟不可见
    up()    # 画笔抬起，移动的时候不画线
    # 将海龟移动到这个位置
    goto(x, y)
    # 画笔落下 -- 移动时将画线
    down()
    color("red", color_name)
    begin_fill()
    forward(size) #画直线
    left(90) #转90度
    forward(size)
    left(90)
    forward(size)
    left(90)
    forward(size)
    left(90)
    end_fill()
    pass
if __name__ == '__main__':  #代码运行的位置
    
    n=4 #贪吃蛇起始
    setup(800, 800) # 设置主窗口的大小和位置
    tracer(False)    # 禁用海龟动画
    for i in range(n):
        if i==n-1:
            draw_square(i*20, 0, 20, 'yellow')
        else:
            draw_square(i*20, 0, 20, 'black')
    
    done()    # 必须作为一个海龟绘图程序的结束语句
